package com.code.test.first.backtracking;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * https://github.com/youngyangyang04/leetcode-master/blob/master/problems/0077.%E7%BB%84%E5%90%88.md
 * 给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。
 * <p>
 * 示例: 输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
 */
public class Code77 {

    public static void main(String[] args) {
        List<List<Integer>> ret = combine2(4, 4);
        System.out.println(ret);
    }

    static List<List<Integer>> ret = new ArrayList<>();
    static LinkedList<Integer> path = new LinkedList<>();

    public static List<List<Integer>> combine(int n, int k) {
        backtracking(n, k, 1);
        return ret;
    }


    public static void backtracking(int n, int k, int startIndex) {
        //每一次采集结果的数量与要求的组合数相等
        if (path.size() == k) {
            ret.add(new ArrayList<>(path));
            return;
        }
        for (int i = startIndex; i <= n; i++) {
            //把自己加进去
            path.add(i);
            //然后从剩下的取一个数，再往下，发现符合组合要求，则放到结果集
            backtracking(n, k, i + 1);
            //再把自己移除，比如 [1,2]→[1]→ [1,3]→[1]→[1,4]
            path.removeLast();
        }
    }

    public static List<List<Integer>> nfor(int n, int k) {
        List<List<Integer>> ret = new ArrayList<>();

        for (int i = 1; i < n; i++) {
            //从大集合里面取1个

            for (int j = i + 1; j < n; j++) {
                //从剩下的取1个

                for (int x = j + 1; x <= n; x++) {
                    List<Integer> cur = new ArrayList<>();
                    cur.add(i);
                    cur.add(j);
                    //从剩下的再取一个
                    cur.add(x);
                    ret.add(cur);
                }
            }
        }
        return ret;
    }

    public static List<List<Integer>> combine2(int n, int k) {
        back(n, k, 1);
        return ret;
    }

    public static void back(int n, int k, int idx) {
        if (path.size() == k) {
            //放进去
            ret.add(new ArrayList<>(path));
            return;
        }
        for (int i = idx; i <= n; i++) {
            path.add(i);
            back(n, k, i + 1);
            path.removeLast();

        }
    }

}
